Color selection and puzzle function fixed