added if statement for companies