Bring back partition and mount modules
Update partition path after partition has been created

Fixes #56