Code fix & add new labels