fix indentation problem
add preliminary assert functionality