fixed small error and added test for Size()