if statement added back