Added controller code but still does not work