Add support for dataframes. Fixes #8