pub fn square(num: i32) -> i32 {
    r#num * num
}