[#6]: Add implementation for test classes