Wrapped up initial code