Cleaned up the code
Updated the readme