Continued work on FYS-KJM4480 midterm 2