Cleaned up Code a bit
Cleaned up git repo