update controller and index to pass second test