Adds some code

Please merge, @mzaher