Updated init code