fixed code and added another line of code