Set colours and add pick a square animation