Fixed up bug in docs and renamed class properly