Add Python answer for first question