Modify to make use of layout.erb