Add special stuff for python