Initial attempt at assignment - both functions complete