Added back If Statement