Adding test for generate_numbers