Updated onclose action