Add three problems code