Messed around
Added test class