Assignment 1: Updated Balance method