I have been looking into expanding the Arduino capabilities. From what I have read the Atmega 164P/324P/644P would best choice. The IO is same for all three but the flash size is different. The board layout would be similar to the current one and you gain the flash size as well as more io. The XX4P models have more features than the standard 32 as well.