Implement tests for certain functionalities