Added second step of third assignment