cleaned code and changed prompt formatting