improve checking answer from user