added another if and else if