P3 of programming assignment done
I think I have to test this again