Small changes... Prettier now