Extend take_screenshots to be able to login