Added init code, tested and confirmed that it works