refactored css classes, thanks @cnvo