Feature/update stock graph with stock balance