Cleaned up and refactored

Bort med struntkod