added more functions for the UI