Unit tests was added