added assosiations to models