Updated the README
Cleaned up the code a bit