Set correct module for exercise 5 assigment 2