Remove test code from lib.rs