add basic functionality in python