Add additional functionality and styles