Cleaned code, better readme.