Separate code and add css