created the if statement to fix the error and init lecture 6
DON!!!!!!!