Correction bugs
Fixed bug #14