Update .onLoad and add .onUnload