Improved comments and readme