Add tests for payment class