Bugfix: margins and target_dim names
Fixes a big introduced in !35 (merged) in lines 288 and 373 (using names for data instead of margins and target_dims, respectively)
Fixes a big introduced in !35 (merged) in lines 288 and 373 (using names for data instead of margins and target_dims, respectively)