Add tests for classes
implementation of computer