added answer for problem 001
added python solution for 004