minor code changes made while going over the solution video