we cahnged the if statement into a unit block