Added code and making things work!