added prompts to make it easier and working on a validation