Improved error checks and smaller dataset.pkl files

Added a coupe of checks that will now through Exceptions with incorrect inputs. Displacements are now saved as int16, just makes the files smaller, shouldn't have an effect otherwise.