Cleaned up readme and comments