work on unit testing and user class