Styling improvements, renamed test file
Adding screenshot, tidying up comments before submitting homework