Add Python solutions for first two problems