Cleanup, and added to readme