improve code +fix responsive