Added functionality to print a victory message, added extra tests