Added some code :)

Kändes som en lagom liten utmaning nu i mellandagarna :)