Cleaned prompt function