Added test implementation + Main class