Made code prettier