# Start writing code here... import numpy as np