Add return statement to initialise function
Update the citation
Implement using `initialise` function