fix layout, add colors to squares