Added reference in the code to the github repository