In this talk we describe some progress on the following problem of A. Sarkozy. For any integer 
, let 
 be the smallest integer such that when the set of squares is coloured in one of 
 colours, every sufficiently large integer can be written as a sum of at most 
 squares. Also, let 
 be the corresponding integer in the analogous context for the set of primes. The problem is to find optimal upper bounds for 
 and 
 in terms of 
.