Some cleanup and added info to readme