Implemented a lose screen (kinda) and also cleaned up a little.