added normalization and divided the dataset into Training/Testing sets