refactoring, simplyfing, working BelongsTo relation