My code for 3331