Update prompt and comments