Added call to keep screen alive