cleaned up code and updated readme