Adding tests to Album module