added code from the next video
followed if and else video