Cleaned up comment and readme