can we remove the hardcoded labels and make them translateable instead when refactoring this?