Some minor error in the package and main file