Fixed Lab 2 B Functionality bug