Update

cleaned up some code, added some to readme