PGM assignment 5