Made code look a bit nicer