improved comments and readme