Linear programming in python