Solution Problem 1 in Python

Signed-off-by: Vishal Thanki <vishalthanki@gmail.com>