Added screen lock feature