fixing figure sizes same as in python