Added extra check if get_current_screen() function exists. Fixes #21