Second ttempt for course project