Update main.rs

implement stock