Lab 8 (kinda working)

@TODO Complete the assignment