Small fix: Menu wouldn't stop for wrong choice and now it does