finished lab

@jackiescool wasn't quite able to figure out return