Added the functionality to the enrol button