cleaning up code
Merge pull request #5 from BraydenCleary/display_data

cleaning up code