Added callbacks and a method to get the active screen