Add main python script