Changes as per Prof comments.

Updated the files according to the comments given by Dr. Beck and added
some test files for future use.