Enhanced the prompt a bit