added class. started to work on #4