Almost completed first assignment