Basics of If statements in python