Use current_screen instead of a body class...more reliable