Cleaned up code and added README