Added unit test based around the given prompt