Almost done with first part of lab 4