Added function to change prompt