Code is outputting/working fine need to add test code