Made code slightly prettier