Added spec for class