Change to init function