added loop with choice to repeate or continue