Add functionality with main view