diff --git a/README.md b/README.md
index 1163c39..b9fc734 100644
--- a/README.md
+++ b/README.md
@@ -1,32 +1,43 @@
# pridefetch 🏳️⚧️
+
Python fetch script for showing your favourite pride flags & system info!
Forked from [megabytesofrem/pridefetch](https://github.com/megabytesofrem/pridefetch).
**Note:** Minimum `Python 3.7` required
## Examples
+
#### Display a trans flag
+
`pridefetch -f trans`
#### Display either a trans or lesbian flag, with a 50/50 chance
+
`pridefetch -r trans,lesbian`
#### List all the available flags
+
`pridefetch -l`
#### Display help with pridefetch
+
`pridefetch -h`
## Quickstart
+
```bash
git clone https://github.com/SpyHoodle/pridefetch.git
cd pridefetch
chmod +x pridefetch
```
+
Then, run pridefetch
+
```bash
./pridefetch
```
+
You can also add pridefetch to your `$PATH` to run it anywhere
+
```bash
mv pridefetch /usr/bin/
```
diff --git a/packages.py b/packages.py
index a0cf4b4..6b1c410 100644
--- a/packages.py
+++ b/packages.py
@@ -25,4 +25,4 @@ def get_num_packages() -> (int, bool):
try:
return len(check_output(["rpm", "-qa"]).decode("utf-8").split("\n")) - 1
except FileNotFoundError:
- return False
\ No newline at end of file
+ return False
diff --git a/pridefetch b/pridefetch
index 4b1b56e..3965027 100755
--- a/pridefetch
+++ b/pridefetch
@@ -2,7 +2,6 @@
from argparse import ArgumentParser
from datetime import timedelta
-from packages import get_num_packages as packages
from getpass import getuser
# from platform import machine as architecture
from platform import platform as system
@@ -10,8 +9,11 @@ from platform import release as kernel
from random import choice as random_choice
from socket import gethostname
from time import clock_gettime, CLOCK_BOOTTIME
+
from distro import name as distribution
+from packages import get_num_packages as packages
+
# Define a dictionary of all the flags and their colors
# Each color is the color for an individual row in the flag
flags = {
@@ -20,7 +22,7 @@ flags = {
"lesbian": [202, 209, 255, 255, 168, 161],
"pansexual": [198, 220, 39],
"trans": [81, 211, 255, 211, 81],
- "nonbinary": [226, 255, 98, 237,],
+ "nonbinary": [226, 255, 98, 237, ],
"demiboy": [244, 249, 117, 255, 117, 249, 244],
"demigirl": [244, 249, 218, 255, 218, 249, 244],
"genderfluid": [211, 255, 128, 0, 63],