Add js changes, update controller