square pattern in python