fix omega from Hz to Rad
just modify the end condition and output