http://github.com/mccoyn/reprap-firmware-fork
I merged my changes (which includes support for the ATMega328) with
another users which includes support for Arduino Mega, so this
firmware should work on both. Look in parameters.h to configure it
for each specific board.
Whenever I make changes to my firmware, I'll update the code on git
hub so it should always be available.