def square_returns_none(x: DataPoint) -> DataPoint:
    if x.num == 2:
        return None
    x.num = x.num**2
    return x
def square(x: DataPoint) -> DataPoint:
    x.num = x.num**2
    return x