Replace square.attr call with removeClass/addClass
Partial progress on score animations