Develop the code for Problem 001