Move if inside initialize function. #3