Adding the data/ directory and comments in the main code.
Comments added to the code and indentation mistakes fixed