Edited the Class per the readme