Add comments, clean up files, edit readme