add solution for problem 1
add even fancier output