Add tests for Machine class