Fix comments and readme