Fixed the sketch, ready for testing