cleaning code, removing grid setup to own file