Modified code so that we did not have to change private functions.
Used setAccessible method.
Cleaned up code and put in some print statements for tests.