Added code for the methods added, changed get to display, added test code