add solution to first problem