add puzzle code