Cleaned up code, readme.