Additional code from Lab5