Cleaned code. Updated readme.