Safe_input lab from session 4, part 2