Add puzzle mechanic