Managed to make the ui work ..