Add if conditionals
Make `if` only accepts bools
Add head + tail
Move code to src folder