Added logic for the number 2