Wrote a decent readme and cleaned up code