Fixes

• Check if function exists for get_current_screen to avoid a fatal if not.