Update 0.5.1

Fixed various issues