Adding Test (& Code that passes it)