Update main to look a little nicer