Add init function to be run on startup