Fix for issue #54

- Replaced calls to .attr with .data