Adding tests for basic functions