Fixed last part of lab2.