updated it to use a grid system