Added get_width & get_height, fixed unit test