Added solution to prob_001