Add styles and logic