Adding "ask for try again" feature into the main