Add number_in_month function