Really dding code from lab 2