added some checks to edit treble