Improved fish_prompt function