Slight bug fixes have been made