Fix indentation + update comment