Fixing prompt output