if statement, added :-)