Top level module (lab5) updated