Commit 2 - adding code and function; testing