Cleaned code and output. Made prompt better.