Ocean Waves exercises in Python