Fixed small image bug
Swift demo view controller