changed ans to num0 since it is nicer (num0, num1) and clean-up

Signed-off-by: Liew Zhao Wei <zhaoweiliew@gmail.com>