put label and if into separate classes