Added puzzle class