Refactored and made add and display functional. Both wokring