Refactor: 3 layouts from 1