dev: solution to problem 1