clean up code, add ideas to readme