Add more basic problem solutions in python