Adding get_label() functionality