Updated podspec to 0.5.4
Change weak reference to assign for backwards compatibility