if statement re-added