Add check_to_quit function