Lab 1 functionality finished