Getting add to work with new layout