Learning swift for fun :)