Bug fixes

#25