some issues with associations... but it works