Minor improvement
#31 I just can't save the labels image... :s