Add back in my previous additions to overlay