Minor updates to comments and readme