Added extra check if get_current_screen() function exists

Fixes #21