Add if-statement