Square Numbers Python