Edited comment position on if-else