wip

- The script should be cleaned up!
- Change actual implemention to a class