Add unit tests and update grunt to 0.4