Cleaned up script for working with dataset