Still need to change partiton sizes