Cleaned up comments and readme