Add companyId to User entity