Adding tests for package module