Loading and preparing the PTB dataset