Add main.rs for simple testing