Added more columns and make grid statefull. Close #11.