Updated the main class to work with the new system approach
Fixed error in argument for lockScreen