Added team problem solution