make better body class function even better