(feat) complete some toy problems