HW 5

Ready to add search