Updated bug UI to use callbacks