modified readme and cleaned up code