separated class from test
separated out test into different methods