The code for the first mini-assignment