Added code to the code stubs provided within the forked repository