wip - adding associations