Fix code prompt formatting