Header button works + small JS fixes