Improved the functionnement and fixed bugs