Made code more user friendly... See header file