Added te last tests and removed a bug. Now only balance remains.