Added problem1.py

Signed-off-by: Jacky Boen <aqiank@gmail.com>