Forgot to add these layout changes. Everything should work now.