Updated init() function