Small changes to Klotski