small update
made class