Added more balance code