How could I optimise this simple python pygame code

Question

I\'ve been set a challenge to make a game using pygame ( I am making snake so it should be easy... but I\'ve never used pygame) so using a more efficient language isn\'t an