Add more code. Don't understand why the layout look this way