Made changes to company generation