Made code look a lot nicer