Added .swift extension

Sorry about that