Add some code for view