edit readme, clean up code