Make code more Swift-style

Made code a bit more idiomatic in Swift-way.
В наследовании от NSObject нету необходимости.