Cleaned up the code. made the UI a bit nicer.
Added new files.