Added check for screenshot dir