Add i3 support