Finished coding part of lab #5