cleaned up code, added text to readme