Creating 40 squares and fixing tests