Just styles fix. I did not touch on the logic.