Roy Model in Python