added weight solution - other possible approach could be to add the weight as... Closes #34 See merge request !47