Added custom body class function