Created a test function. File is compiling but have not run. Need to add some plotting functionality