Write tests for functions
Write tests for functions