Added tree problems in python.

Signed-off-by: Ionut Radu <ihonius@gmail.com>