Handled edge cases, stylistic changes