deleted a comment
added activation function
Added activation function and main code