Cleaned up code a little bit. Only to check how branches work in Git.