Now got balance working