Clean code, add overlay div