Edited checkBalance