Add some style and logics