improved your code
hihi