Cleaned up code and added README.md