Update jQuery context and fix css