Moved layout from init to layoutSubviews, fixes #4