Add short skip connection and CBAM module