after revising my code with udacity.