Added socket listener
Added getBalance method